coding exercise for beginners in python with solution